home *** CD-ROM | disk | FTP | other *** search
/ Total Network Tools 2002 / NextStepPublishing-TotalNetworkTools2002-Win95.iso / Archive / Misc Servers / Zope.exe / RDB.PY < prev    next >
Encoding:
Python Source  |  2000-07-17  |  10.0 KB  |  285 lines
  1. ##############################################################################
  2. # 
  3. # Zope Public License (ZPL) Version 1.0
  4. # -------------------------------------
  5. # 
  6. # Copyright (c) Digital Creations.  All rights reserved.
  7. # 
  8. # This license has been certified as Open Source(tm).
  9. # 
  10. # Redistribution and use in source and binary forms, with or without
  11. # modification, are permitted provided that the following conditions are
  12. # met:
  13. # 
  14. # 1. Redistributions in source code must retain the above copyright
  15. #    notice, this list of conditions, and the following disclaimer.
  16. # 
  17. # 2. Redistributions in binary form must reproduce the above copyright
  18. #    notice, this list of conditions, and the following disclaimer in
  19. #    the documentation and/or other materials provided with the
  20. #    distribution.
  21. # 
  22. # 3. Digital Creations requests that attribution be given to Zope
  23. #    in any manner possible. Zope includes a "Powered by Zope"
  24. #    button that is installed by default. While it is not a license
  25. #    violation to remove this button, it is requested that the
  26. #    attribution remain. A significant investment has been put
  27. #    into Zope, and this effort will continue if the Zope community
  28. #    continues to grow. This is one way to assure that growth.
  29. # 
  30. # 4. All advertising materials and documentation mentioning
  31. #    features derived from or use of this software must display
  32. #    the following acknowledgement:
  33. # 
  34. #      "This product includes software developed by Digital Creations
  35. #      for use in the Z Object Publishing Environment
  36. #      (http://www.zope.org/)."
  37. # 
  38. #    In the event that the product being advertised includes an
  39. #    intact Zope distribution (with copyright and license included)
  40. #    then this clause is waived.
  41. # 
  42. # 5. Names associated with Zope or Digital Creations must not be used to
  43. #    endorse or promote products derived from this software without
  44. #    prior written permission from Digital Creations.
  45. # 
  46. # 6. Modified redistributions of any form whatsoever must retain
  47. #    the following acknowledgment:
  48. # 
  49. #      "This product includes software developed by Digital Creations
  50. #      for use in the Z Object Publishing Environment
  51. #      (http://www.zope.org/)."
  52. # 
  53. #    Intact (re-)distributions of any official Zope release do not
  54. #    require an external acknowledgement.
  55. # 
  56. # 7. Modifications are encouraged but must be packaged separately as
  57. #    patches to official Zope releases.  Distributions that do not
  58. #    clearly separate the patches from the original work must be clearly
  59. #    labeled as unofficial distributions.  Modifications which do not
  60. #    carry the name Zope may be packaged in any form, as long as they
  61. #    conform to all of the clauses above.
  62. # 
  63. # 
  64. # Disclaimer
  65. # 
  66. #   THIS SOFTWARE IS PROVIDED BY DIGITAL CREATIONS ``AS IS'' AND ANY
  67. #   EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  68. #   IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  69. #   PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL DIGITAL CREATIONS OR ITS
  70. #   CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  71. #   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  72. #   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
  73. #   USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  74. #   ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  75. #   OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
  76. #   OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  77. #   SUCH DAMAGE.
  78. # 
  79. # 
  80. # This software consists of contributions made by Digital Creations and
  81. # many individuals on behalf of Digital Creations.  Specific
  82. # attributions are listed in the accompanying credits file.
  83. # 
  84. ##############################################################################
  85. __doc__='''Class for reading RDB files
  86.  
  87.  
  88. $Id: RDB.py,v 1.24.32.2 2000/07/17 15:03:55 brian Exp $'''
  89. __version__='$Revision: 1.24.32.2 $'[11:-2]
  90.  
  91. import regex, regsub
  92. from string import split, strip, lower, upper, atof, atoi, atol, find, join
  93. import DateTime
  94. from Missing import MV
  95. from array import array
  96. from Record import Record
  97. from Acquisition import Implicit
  98. import ExtensionClass
  99.  
  100. def parse_text(s):
  101.     if find(s,'\\') < 0 and (find(s,'\\t') < 0 and find(s,'\\n') < 0): return s
  102.     r=[]
  103.     for x in split(s,'\\\\'):
  104.         x=join(split(x,'\\n'),'\n')
  105.         r.append(join(split(x,'\\t'),'\t'))
  106.     return join(r,'\\')
  107.  
  108.  
  109. Parsers={'n': atof,
  110.          'i': atoi,
  111.          'l': atol,
  112.          'd': DateTime.DateTime,
  113.          't': parse_text,
  114.          }
  115.  
  116. class SQLAlias(ExtensionClass.Base):
  117.     def __init__(self, name): self._n=name
  118.     def __of__(self, parent): return getattr(parent, self._n)
  119.  
  120. class NoBrains: pass
  121.  
  122. class DatabaseResults:
  123.     """Class for reading RDB files
  124.     """
  125.     _index=None
  126.  
  127.     # We need to allow access to not-explicitly-protected
  128.     # individual record objects contained in the result.
  129.     __allow_access_to_unprotected_subobjects__=1
  130.  
  131.     def __init__(self,file,brains=NoBrains, parent=None, zbrains=None):
  132.  
  133.         self._file=file
  134.         readline=file.readline
  135.         line=readline()
  136.         self._parent=parent
  137.         if zbrains is None: zbrains=NoBrains
  138.  
  139.         comment_pattern=regex.compile('#')
  140.         while line and comment_pattern.match(line) >= 0: line=readline()
  141.  
  142.         line=line[:-1]
  143.         if line and line[-1:] in '\r\n': line=line[:-1]
  144.         self._names=names=split(line,'\t')
  145.         if not names: raise ValueError, 'No column names'
  146.  
  147.         aliases=[]
  148.         self._schema=schema={}
  149.         i=0
  150.         for name in names:
  151.             name=strip(name)
  152.             if not name:
  153.                 raise ValueError, 'Empty column name, %s' % name
  154.             if schema.has_key(name):
  155.                 raise ValueError, 'Duplicate column name, %s' % name
  156.             schema[name]=i
  157.             n=lower(name)
  158.             if n != name: aliases.append((n, SQLAlias(name)))
  159.             n=upper(name)
  160.             if n != name: aliases.append((n, SQLAlias(name)))
  161.             i=i+1
  162.  
  163.         self._nv=nv=len(names)
  164.         line=readline()
  165.         line=line[:-1]
  166.         if line[-1:] in '\r\n': line=line[:-1]
  167.         
  168.         self._defs=defs=split(line,'\t')
  169.         if not defs: raise ValueError, 'No column definitions'
  170.         if len(defs) != nv:
  171.             raise ValueError, (
  172.                 """The number of column names and the number of column
  173.                 definitions are different.""")
  174.         
  175.         i=0
  176.         self._parsers=parsers=[]
  177.         defre=regex.compile('\([0-9]*\)\([a-zA-Z]\)?')
  178.         self._data_dictionary=dd={}
  179.         self.__items__=items=[]
  180.         for _def in defs:
  181.             _def=strip(_def)
  182.             if not _def:
  183.                 raise ValueError, ('Empty column definition for %s' % names[i])
  184.             if defre.match(_def) < 0:
  185.                 raise ValueError, (
  186.                     'Invalid column definition for, %s, for %s'
  187.                     % _def, names[i])
  188.             type=lower(defre.group(2))
  189.             width=defre.group(1)
  190.             if width: width=atoi(width)
  191.             else: width=8
  192.  
  193.             try: parser=Parsers[type]
  194.             except: parser=str
  195.  
  196.             name=names[i]
  197.             d={'name': name, 'type': type, 'width': width, 'parser': parser}
  198.             items.append(d)
  199.             dd[name]=d
  200.             
  201.             parsers.append((i,parser))
  202.             i=i+1
  203.  
  204.         # Create a record class to hold the records.
  205.         names=tuple(names)
  206.  
  207.         class r(Record, Implicit, brains, zbrains):
  208.             'Result record class'               
  209.  
  210.         r.__record_schema__=schema
  211.         for k in filter(lambda k: k[:2]=='__', Record.__dict__.keys()):
  212.             setattr(r,k,getattr(Record,k))
  213.  
  214.         # Add SQL Aliases
  215.         d=r.__dict__
  216.         for k, v in aliases:
  217.             if not hasattr(r,k): d[k]=v
  218.  
  219.         if hasattr(brains, '__init__'):
  220.             binit=brains.__init__
  221.             if hasattr(binit,'im_func'): binit=binit.im_func
  222.             def __init__(self, data, parent, binit=binit):
  223.                 Record.__init__(self,data)
  224.                 binit(self.__of__(parent))
  225.  
  226.             r.__dict__['__init__']=__init__
  227.                     
  228.  
  229.         self._class=r
  230.  
  231.         # OK, we've read meta data, now get line indexes
  232.  
  233.         p=file.tell()
  234.         save=self._lines=array('i')
  235.         save=save.append
  236.         l=readline()
  237.         while l:
  238.             save(p)
  239.             p=p+len(l)
  240.             l=readline()
  241.  
  242.     def _searchable_result_columns(self): return self.__items__
  243.     def names(self): return self._names
  244.     def data_dictionary(self): return self._data_dictionary
  245.  
  246.     def __len__(self): return len(self._lines)
  247.  
  248.     def __getitem__(self,index):
  249.         if index==self._index: return self._row
  250.         file=self._file
  251.         file.seek(self._lines[index])
  252.         line=file.readline()
  253.         line=line[:-1]
  254.         if line and line[-1:] in '\r\n': line=line[:-1]
  255.         fields=split(line,'\t')
  256.         l=len(fields)
  257.         nv=self._nv
  258.         if l != nv:
  259.             if l < nv:
  260.                 fields=fields+['']*(nv-l)
  261.             else:
  262.                 raise ValueError, (
  263.                     """The number of items in record %s is invalid
  264.                     <pre>%s\n%s\n%s\n%s</pre>
  265.                     """ 
  266.                     % (index, ('='*40), line, ('='*40), fields))
  267.         for i, parser in self._parsers:
  268.             try: v=parser(fields[i])
  269.             except:
  270.                 if fields[i]:
  271.                     raise ValueError, (
  272.                         """Invalid value, %s, for %s in record %s"""
  273.                         % (fields[i], self._names[i], index))
  274.                 else: v=MV
  275.             fields[i]=v
  276.  
  277.         parent=self._parent
  278.         fields=self._class(fields, parent)
  279.         self._index=index
  280.         self._row=fields
  281.         if parent is None: return fields
  282.         return fields.__of__(parent)
  283.  
  284. File=DatabaseResults
  285.